﻿#pragma once
#include<iostream>
#include<vector>
using namespace std;
// 链地址法的实现
namespace shuaiming_hash_bucket
{
	template<class K>
	struct KeyOfTFunc1
	{
		const K& operator()(const K& key)
		{
			return key;
		}
	};
	template<class K, class V>
	struct KeyOfTFunc2
	{
		const K& operator()(const pair<K, V>& kv)
		{
			return kv.first;
		}
	};
	template<class K>
	class HashFunc
	{
	public:
		size_t operator()(const K& key)
		{
			return (size_t)key;
		}
	};

	// 哈希表中支持字符串的操作
	template<>
	class HashFunc<string>
	{
	public:
		size_t operator()(const string& key)
		{
			size_t hash = 0;
			for (auto e : key)
			{
				hash *= 131;
				hash += e;
			}
			return hash;
		}
	};
	template<class T>
	struct HashNode
	{
		T _data;
		HashNode<T>* _next;
		HashNode(const T& data)
			:_data(data)
			, _next(nullptr)
		{}
	};
	// K 为 T 中key的类型
	// T 可能是键值对，也可能是K
	// KeyOfT: 从T中提取key
	// Hash将key转化为整形，因为哈希函数使用除留余数法
	template<class K, class T, class KeyOfT, class Hash = HashFunc<K>>
	class HashTable
	{

		typedef HashNode<T> Node;
	public:
		HashTable()
		{
			_tables.resize(10, nullptr);
		}
		size_t size() const { return _n; }
		// 哈希桶的销毁
		~HashTable()
		{
			for (int i = 0; i < _tables.size(); i++)
			{
				Node* cur = _tables[i];
				while (cur)
				{
					Node* next = cur->_next;
					delete cur;
					cur = next;
				}
				_tables[i] = nullptr;
			}
		}

		// 插入值为data的元素，如果data存在则不插入
		bool Insert(const T& data)
		{
			if (Find(kt(data)))
				return false;
			size_t hash0 = hs(kt(data)) % _tables.size();
			// 扩容
			if (_n == _tables.size())
			{
				vector<Node*> newHashTable(_tables.size() * 2);
				for (int i = 0; i < _tables.size(); i++)
				{
					Node* cur = _tables[i];
					while (cur)
					{
						size_t newhash0 = hs(kt(cur->_data)) % newHashTable.size();
						Node* next = cur->_next;
						cur->_next = newHashTable[newhash0];
						newHashTable[newhash0] = cur;
						cur = next;
					}
					_tables[i] = nullptr;
				}
				newHashTable.swap(_tables);
			}
			// 头插
			Node* newnode = new Node(data);
			newnode->_next = _tables[hash0];
			_tables[hash0] = newnode;
			++_n;
			return true;
		}

		// 在哈希桶中查找值为key的元素，存在返回true否则返回false﻿
		bool Find(const K& key)
		{
			size_t hash0 = hs(key) % _tables.size();
			Node* cur = _tables[hash0];
			while (cur)
			{
				if (kt(cur->_data) == key)
					return true;
				cur = cur->_next;
			}
			return false;
		}

		// 哈希桶中删除key的元素，删除成功返回true，否则返回false
		bool Erase(const K& key)
		{
			size_t hash0 = hs(key) % _tables.size();
			Node* cur = _tables[hash0];
			while (cur)
			{
				if (kt(cur->_data) == key)
				{
					Node* next = cur->_next;
					if (cur == _tables[hash0])
					{
						_tables[hash0] = next;
					}
					else
					{
						_tables[hash0]->_next = next;
					}
					delete cur;
					--_n;
					return true;
				}
				cur = cur->_next;
			}
			return false;
		}

	private:
		vector<Node*> _tables;  // 指针数组
		size_t _n = 0;			// 表中存储数据个数
		KeyOfT kt;
		Hash hs;
	};
}